package com.css.zfzx.fxpg.modules.vulnerability.area.controller;

import com.alibaba.fastjson.JSONObject;
import com.css.bpm.platform.base.response.RestResponse;
import com.css.zfzx.fxpg.common.utils.DownloadUtils;
import com.css.zfzx.fxpg.modules.vulnerability.area.repository.entity.FxVulnerabilityAreaEntity;
import com.css.zfzx.fxpg.modules.vulnerability.area.service.FxVulnerabilityAreaService;
import com.css.zfzx.fxpg.modules.vulnerability.area.viewobjects.FxVulnerabilityAreaQueryParams;
import com.css.bpm.platform.utils.PlatformPageUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;

/**
 * @author leon
 * @date 2020-11-10
 */
@Slf4j
@RestController
@RequestMapping("/fx/fxVulnerabilityAreas")
public class FxVulnerabilityAreaController {
    @Autowired
    private FxVulnerabilityAreaService fxVulnerabilityAreaService;
    @Autowired
    DownloadUtils downloadUtils;

    @GetMapping("/queryFxVulnerabilityAreas")
    public RestResponse queryFxVulnerabilityAreas(HttpServletRequest request, FxVulnerabilityAreaQueryParams queryParams) {
        RestResponse response = null;
        try {
            int curPage = Integer.valueOf(request.getParameter(PlatformPageUtils.CUR_PAGE));
            int pageSize = Integer.valueOf(request.getParameter(PlatformPageUtils.PAGE_SIZE));
            JSONObject jsonObject = fxVulnerabilityAreaService.queryFxVulnerabilityAreas(queryParams, curPage, pageSize);
            response = RestResponse.succeed(jsonObject);
        } catch (Exception e) {
            String errorMessage = "查询失败!";
            log.error(errorMessage, e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    @GetMapping("/listFxVulnerabilityAreas")
    public RestResponse listFxVulnerabilityAreas(FxVulnerabilityAreaQueryParams queryParams) {
        RestResponse response = null;
        try {
            List<FxVulnerabilityAreaEntity> list = fxVulnerabilityAreaService.listFxVulnerabilityAreas(queryParams);
            response = RestResponse.succeed(list);
        } catch (Exception e) {
            String errorMessage = "查询失败!";
            log.error(errorMessage, e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    @GetMapping("{id}")
    public RestResponse getFxVulnerabilityArea(@PathVariable String id) {
        RestResponse response = null;
        try {
            FxVulnerabilityAreaEntity fxVulnerabilityArea = fxVulnerabilityAreaService.getFxVulnerabilityArea(id);
            response = RestResponse.succeed(fxVulnerabilityArea);
        } catch (Exception e) {
            String errorMessage = "获取失败!";
            log.error(errorMessage, e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    @PostMapping
    public RestResponse saveFxVulnerabilityArea(@RequestBody FxVulnerabilityAreaEntity fxVulnerabilityArea) {
        RestResponse response = null;
        JSONObject json = new JSONObject();
        try {
            fxVulnerabilityAreaService.saveOrUpdate(fxVulnerabilityArea);
            json.put("message", "新增成功!");
            response = RestResponse.succeed(json);
        } catch (Exception e) {
            String errorMessage = "新增失败!";
            log.error(errorMessage, e);
            response = RestResponse.fail(errorMessage);
        }
        return response;

    }

    @PutMapping
    public RestResponse updateFxVulnerabilityArea(@RequestBody FxVulnerabilityAreaEntity fxVulnerabilityArea) {
        RestResponse response = null;
        JSONObject json = new JSONObject();
        try {
            fxVulnerabilityAreaService.updateFxVulnerabilityProvinceByArea(fxVulnerabilityArea);
            fxVulnerabilityAreaService.updateFxVulnerabilityArea(fxVulnerabilityArea);
            json.put("message", "修改成功!");
            response = RestResponse.succeed(json);
        } catch (Exception e) {
            String errorMessage = "修改失败!";
            log.error(errorMessage, e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    @DeleteMapping
    public RestResponse deleteFxVulnerabilityAreas(@RequestParam List<String> ids) {
        RestResponse response = null;
        JSONObject json = new JSONObject();
        try {
            fxVulnerabilityAreaService.deleteFxVulnerabilityAreas(ids);
            json.put("message", "删除成功!");
            response = RestResponse.succeed(json);
        } catch (Exception e) {
            String errorMessage = "删除失败!";
            log.error(errorMessage, e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }

    @GetMapping("/getValidDictItemsByDictCode/{dictCode}")
    public RestResponse getValidDictItemsByDictCode(@PathVariable String dictCode) {
        RestResponse restResponse = null;
        try {
            restResponse = RestResponse.succeed(fxVulnerabilityAreaService.getValidDictItemsByDictCode(dictCode));
        } catch (Exception e) {
            String errorMsg = "字典项获取失败!";
            log.error(errorMsg, e);
            restResponse = RestResponse.fail(errorMsg);
        }
        return restResponse;
    }

    /**
     * 导入
     *
     * @param file
     * @return
     */
    @PostMapping("/import")
    public RestResponse importExcel(@RequestParam("file") MultipartFile file) {
        RestResponse restResponse = null;
        try {
            String s = fxVulnerabilityAreaService.importExcel(file);
            restResponse = RestResponse.succeed(s);
        } catch (Exception e) {
            String errorMsg = "导入失败!";
            log.error(errorMsg, e);
            restResponse = RestResponse.fail(errorMsg);
        }
        return restResponse;
    }

    /**
     * 导出
     *
     * @param queryParams
     * @param response
     * @return
     */
    @GetMapping("/export")
    public RestResponse exportExcel(FxVulnerabilityAreaQueryParams queryParams, HttpServletResponse response) {
        RestResponse restResponse = null;
        try {
            fxVulnerabilityAreaService.exportExcel(queryParams, response);
            restResponse = RestResponse.succeed("导入成功!");
        } catch (Exception e) {
            String errorMsg = "导入失败!";
            log.error(errorMsg, e);
            restResponse = RestResponse.fail(errorMsg);
        }
        return restResponse;
    }

    @GetMapping("/initialize")
    public RestResponse initialize() {
        RestResponse response = null;
        JSONObject json = new JSONObject();
        try {
            fxVulnerabilityAreaService.initialize();
            json.put("message", "初始化成功!");
            response = RestResponse.succeed(json);
        } catch (Exception e) {
            String errorMessage = e.getMessage();
            log.error(errorMessage, e);
            response = RestResponse.fail(errorMessage);
        }
        return response;
    }
}